Accordion: Elastic-Depth Inference via Fixed-Prefix Layer Looping
Accordion:通过固定前缀层循环实现弹性深度推理
🧭 LLM 弹性深度推理 · 固定前缀层循环 ★ 深度解读
Tsinghua University · Baidu Inc.
一个 MoE 模型通过固定前缀层循环支持五个推理预算,中间预算以约 21-24% 额外时间捕获 89% 深度增益
📌 工作参考价值(4 点)
- 需要线上不同 latency/cost 档位时,可训练一个模型支持多个推理预算,通过固定前缀层循环共享权重,避免维护多个模型;推理时按请求选择预算,FLOPs 和 KV cache 线性扩展。
- 训练时每个 micro-batch 均匀采样一个预算路径,只优化该路径 NTP,无需跨预算一致性损失;第二遍用 zero-initialized tanh gates,初始为恒等映射,训练稳定。预算无关参数优于 budget-conditioned 参数。
- 多领域能力整合可采用参数组加权合并专家,再做 budget-matched 多教师在线蒸馏(MOPD),能保留各预算增益并提升整体指令遵循,适合搜索推荐中的 Agent 多任务模型合并。
- 资源敏感场景优先选中间预算(如 k=6):以约 21-24% 额外 TPOT 捕获约 89% 深度增益,推荐系统在线服务可按此设计弹性档位。